#pragma once

#include <iostream>
#include <string>
#include <atomic>
#include <vector>
#include <fstream>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/time.h>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    const std::string temp_path = "./temp/";

    class TimeUtil
    {
    public:
        // 返回时间戳
        static std::string GetTimeStamp()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec);
        }

        // 返回毫秒级时间戳
        static std::string GetMsTimeStamp()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec * 1000 + _time.tv_usec / 1000);
        }
    };

    class PathUtil
    {
    public:
        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string path_name = temp_path;
            path_name += file_name;
            path_name += suffix;
            return path_name;
        }

        // 构建源文件路径+添加后缀
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }

        // 构建可执行文件路径+添加后缀
        static std::string Exe(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }

        // 构建程序编译错误时对应的文件路径+添加后缀
        static std::string CompileError(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compile_error");
        }

        // 构建程序运行时标准输入对应的文件路径+添加后缀
        static std::string Stdin(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }

        // 构建程序运行时标准输出对应的文件路径+添加后缀
        static std::string Stdout(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }

        // 构建程序运行错误时对应的文件路径+添加后缀
        static std::string Stderr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
    };

    class FileUtil
    {
    public:
        // 判断对应路径文件是否存在
        static bool IsFileExist(const std::string &path_name)
        {
            // 获取文件属性，获取成功则存在，获取失败则不存在
            struct stat st;
            if (stat(path_name.c_str(), &st) == 0) // 成功
            {
                return true;
            }
            return false; // 失败
        }

        // 构造唯一的文件名
        static std::string UniqFileName()
        {
            static std::atomic_uint id(0);
            id++;
            // 毫秒级时间戳 + 原子递增唯一值
            std::string ms = TimeUtil::GetMsTimeStamp();
            std::string uid = std::to_string(id);
            return ms + "_" + uid;
        }

        // 构造文件并写入内容
        static bool WriteFile(const std::string &target_file, const std::string &content)
        {
            // 打开文件流
            std::ofstream out(target_file);

            // 判断是否打开成功
            if (!out.is_open())
                return false;

            // 写入文件
            out.write(content.c_str(), content.size());

            // 关闭文件流
            out.close();
            return true;
        }

        // 读取文件内容
        // target_file: 绝对路径
        // br: 按行读取内容时是否换行
        static bool ReadFile(const std::string &target_file, std::string *content, bool br = false)
        {
            (*content).clear();
            // 打开文件流
            std::ifstream in(target_file);

            // 判断是否打开成功
            if (!in.is_open())
                return false;

            // 按行读取文件
            std::string line;
            while (std::getline(in, line)) // getline默认读取时去掉\n
            {
                (*content) += line;
                (*content) += (br ? "\n" : ""); // 按需求决定是否添加\n
            }

            // 关闭文件流
            in.close();
            return true;
        }
    };

    class StringUtil
    {
    public:
        // 将目标字符串按照分割符切成多份
        static void SplitString(const std::string &str, std::vector<std::string> *target, const std::string &sep)
        {
            boost::split((*target), str, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };
}